fixing linter issues, adding comments
started using argparser as suggested in #1