Added animation code