Fixing toggle class