Adding test case, adding output

@vishaalprasad if you could add some meaningful test cases to the `__main__` program, that would be great!

Looks like your editor doesn't lint (remove trailing whitespace); if you can turn on a setting in your editor to lint, that would help make the github diffs more meaningful!

Please take a look at the changes and leave comments if you have any questions or concerns!