fixed test cases and should now work