Adding in testing