Added small fixes to styling