main.cpp repaired