Removed code that was unnecessary.
Added one more test case.