Fixing more styling