Made minor changes to code to reduce the number of lines used, and documented the fact that I believe my code works in the README file