Made minor changes to styling