Small update to styling