Add some css classes back in