Update main.cpp

Testing new functions