updated iteration exercise without {}