re-work the css