refactor: minor general refactor and docs improvements