Tests

add a unit test class