minor improvements
update comment