Added simple test cases for part 1