add fit file

need to check if correct values for train