Smoothing out the overfitting.py code
Fixed colors