Separate training 2d plot function