Updated README
Added TODOs from code review with Tom
Probability generator file - untested