Minor code cleanup, and fixed flying motion when you fly around a corner.
Fixed the issue about keeping the direction when you transform back into robot mode.