Fixed exit conditons

Also added B and W files for possible probability diet.

Signed-off-by: Tom Kent <TomeKent@hotmail.com>