Minor style tweaking