First writing of Python task