small code fix
new main file