test works
add more tests