TODO doc clean up