Focus on input after animation completes