Test cases

Test cases written, but still needs some work