Modify Q3.py
Finish the assignments of week4 and week5.