modified to do next predictions and print as ints not float