Changed to generate some test data

changed to generate some test data
tested with diff probabilities

The test is complete. Now need to write details about precision and
recall