Cleaning up generate_types.py script