Tidied up how movement works to make it easier to change direction