Added toggle class function