Added a function that draws the input and output puzzles and tidied up the code.
Getting the readme I added on Github