HW 2

Improved movement