updated the test, including more testing