Minor update to styling