Adding more tests to the simple case