Iteration 4

small changes