Added regression prediction