works for prob 0, 1 and 2 with ArchCost set to 2