Improve python examples