add code comments and cleanup
2 files changed