adding some basic unit testing