add cbb520_hw2_part4 python script