Made some small code style improvements.