Added distribution class tests