Forgot close files

In prob-gen-1.py