fixed to main.cpp