updated comments and docs