add few cosmetic changes