Add class to control animation