cleaning main.cpp