Adds doc and removes puts statement.