think i'm done. now the test cases