Minor modifications to compiler.py