Fixing the styling