Challenge 1 & 2 (no puts and each_with_index)