Update code and example