Some small improvements to styling