Update main.c

Re-define functions created