Even more polish to Python binding.