Add Python example to docs