Решение, по возможности, оформите в виде jupyter notebook.