Fixed errors on main.cpp