Add cosmetic changes