Added some basic unit testing. It needs expansion