Added a couple unit tests as examples