Fixes to styling