Fixing main.cpp

Streamlining main program process is main.cpp