Add logic unit test