Final working version, Python 1 class, Wk 4 assignment