implement first unit tests