mini project now uses prediction