Rotation added, the units are now rotated correctly as well.
(rotate toward the direction)