small updates to p009.py