Setup polygram.py as a class