Add Python example code