Create generator.py, a problem-generator for the lab