changing partial fit to be seperate from _fit