small cosmetic changes in code