Add code for testing API