Adding some tests to get going