Update problemgen.py

Added comments! They're important.