Made some small style changes