WIP generate.py