problem set 0 of MIT OCW - python - Intro to computer science & programming