Quick additional fixes to styling