Remove unnecessary code from example