add cbb520_hw2_Part5 python script