Add ability to move units