Documentation

minor code changes