Updated some test cases

yessir