added a few things to main.cpp