minor changes to main