Improving main.c comments