Added python code in the pyton folder, it should work