WK#2 assignment #1 update to code. Added array and comment.