Removing puts statements