refactor: add comments and todo