add tests to code
add tests to code..