Css needs to be nicer