updated: compinvest1-hwk01-q01.py