adding in testing framework