refactor(main): make the code more testable and easier to read (#1347)

refactor(main): make the code more testable and easier to read (#1347) #2155