I removed it as the signature was hell due to the many Epochs parameters + freqs + reg param etc...