Test cases

Added some test cases in comments