clean up
add unit tests