clean up code and fix example parameter