Write some test for the API