Adding in a testing section