Minor improvements to styling