fixed part two of the diamond exercise