Update assignment3_3.py

My more complex version, final.