Updated docstring for fit()
Remove mock modules