Update main.cpp

Added one more include