Started with actual prediction code