added plotting class