generator.py: Added todo about func return type