#36 adding unit tests