fix up styling a bit more