Changed class to define so the type can be used