Remove unused css class
Update unit test