implemented two tests
minor fix for testcase