Removed unnecessary code from example