Clean up

Cleaned up and commented the code.  Also adding the example outputs