minor additions to code