Update back to original solution

Updated this repo to the original solution. After I cloned and pushed the repo from my local repo to GitHub I changed my answer online to reflect the original requirements of the assignment.