Add animation on the button