working on hw5 python