Updated css to work with current classes