changed test cases to check set key, added 2 more test case to cover old code