rm unnecessary imports
hardcode random distributions for tests