Update main.cpp

Now it uses the primary header