Add preliminary code to fit