Added animation code back in