add a few comments and doc strings