again little modifications

little modifications in main