minor updates in the main file