minor adjustments to code