Add doc strings.  General clean up.