Added some test for Z with Distribution