Started working on predictions, not finished or tested yet