small fix  to iteration 3