small cosmetic code changes