Improve Python docs