Clean up css and classes