add a few comments
small refactor