Minor style udpate