Added update function for the container code