Added simple setup case.
Add few more test cases along with needed functionality.