Added some unit tests to handle some basic use cases.