Got predictions working