Change argument order in fit for compatability, update docstring