Completed 2nd part of the first exercise for control flows and arrays