Fixing up some styling