Pylint polishing
Try to turn this into a proper python package