added confinement probability, but haven't checked it thoroughly