python/grade-school: First working solution