Cleaning main.cpp