starting on sample solution in Python