Refactor as a module with a setup