Fixed removeClass method
Check added to unit tests