added python code correctly

made a .py.txt file before by mistake