Update main.c

No of test cases and 'free' keyword is included!