Forgot to update main.cpp