Add reading from stdin to python