Adding animation code