Improvements to koan runner and minor fixes to tests