Add code for animation