Added more tests to test predictMovement function