Add py program

See above