Added updating from the main class.