some minor modifications to the predictEpsilon example