start adding unit testing