Fixing some styling