Clearer/shorter docstring.
Fixed program; updated example outputs.

Fixed typos/bugs that made the previous version not work!