Increased perfromance of regression prediction function