It must be a valid c/c++ program. Just wrap your code in a function