Improved styling a bit