adding code on main.c