Add first unit tests