made a python class responsible for plotting