Made some small cosmetic changes