Fixed main.cpp