removed assertions and print probabilities
added unit test for sigmoid fitting