Documentation + TODO updates