Updated documentation script and TODO