minor changes to work