pylab not needed?
Move regression.py to test_regression.py to be auto found