Added unit movement