First version working (only plot, not creating classes!)