minor changes to add