finally figured out how to write tests

Adding a couple of sample tests for now