add tests to ensure stability
add tests to ensure stability