fix bug + add unit test folder