main: refactor
main: refactor