Small changes to styling