add first unit tests