update cbb520 hw2 part3 python scripts