Adding matplotlib comment back in