Updated test case
small updates to test case