added first unit tests