Add probability distribution class and tests
Update asserts to have expected first and then actual.