Added basic probability tests