Respond to review feedback and clean up pylint violations