Added risk, but still need to change its loss function