refactoring the css to be simpler