Added method to stop animation