Minor updates to styling