cleaning up for pylint