Added prediction and statistics functionality