Improve Python test program