Fixing up styling