A solution to the problem with making flying mobs.

So I know any developers of mcreator probably wont see this, but if they do, hear me out. I came up with a way for you guys to implement the option to make flying creatures that work. You should try using the flying ai of preexisting mobs. But you should step it from the rest of its ai. An example of this would be a passive mob that flys like a phantom. So there should be different options, like of you want it to fly like a phantom, ghastly, ender dragon, or bat. I am still learning how to code so I dont know if it is really that simple. Thanks.