~

minor changes to code