Fixed the test cases. Should close issue #1.