added basic unit test