minor changes to main file