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.
This PR contains all the changes made in that one, as well as some additional changes that are open for discussion.