Use classes for setup scripts