Add params to fit usecase