Pylint script and add proper __main__