Improved styling a little