Fixed styling a little more