Completed 1st part of the second exercise for control flows and arrays