Add correct plot