minor changes to fix errors