Update README.md

return statement added to python example.