Changed test cases to use 3 lines, added more test cases