Change snippet for main() for C and C++