Added mock file
modified mock to create useful test cases