Added animation functionality