Fixed tests, and added Y axis movement