Shortening unit rotation code

Signed-off-by: Tim Ullrich <trozen@gmail.com>