minor update
refactor