Added additional function arguments to custom estimation function