Added return to movement