Added function main to main.cpp

(to test again)