start adding unit tests