Add code for animation of move