Made the css a bit less generic