added return statement to main.cpp