Minor update to testing file