Adding save controller