Add marshmallow integration