Test cases

Added test cases, psuedo code