Решение этой задачи, а также многие другие дополнительные ресурсы доступны в интернете по адресу realpython.com/python-basics/resources