Wrap it in a module instead of class