Helen's poor attempt with Python