Add update functionality throughout the system, where applicable