Fixed up the css slightly