added basic unit tests