Update functionality working