Add test functions to repo