Started Python solution month 3