add tests for fit