Changes in main problem in Arrays lab

Here is a rough draft of the changes suggested in https://github.com/byuimpact/numerical_computing/pull/102

We should be able to merge this by itself, or after merging the other one.