Added a line to main.cpp