Update task doc gen script

Docs are now fetched with Code.get_docs/2