Got some simple probability-based tests passing