Create one setup function rather than two sections of identical code