Added some test cases.

If you want me to create the tests directory and move the test case there, just let me know.