This may be an issue if the class doesn't have units and the user passes a unit. Not sure what I should do here.