Add two test functions