Added comments

The code was very succinct and readable.  I just added a few suggestions for minor enhancements.