change things
add unit test