adding initial unit tests