added parameter to fit