update HW0.py and add map