Could use slight modifications

The program is largely commented and functions as intended, however,
there is some room for extra error handling and/or minor enhancements.