Adding update function for volume