Using API class for running tests