Refactor the css