updating test in viz and notebook