Corrected add remove class.