Added todos
New tests for diff count vs failure probability