main: add tests