TODO updated
cosmetic fixes