forgot update to main.cpp