Make sure all test cases pass (#26)