It worked! Adding update of context