Minor changes on main.c