Update main.cpp

now includes basic input/output code