Updated plot to accomodate subplot