saving predictions feature added