Add wrap method