little more modifications

little modifications on main