added ability to do tests