Change name of test function and updated class diagram for API