Update functionality working again