First exercise in Py