minor changes to mainfile