add tests to test folder