Add option staypuft in marshmallow