added tests to better verify working correctly