Add custom marshal_with decorator