prediction working slightly