added more data prediction and testing