Adds some basic test cases
Initial implementation