Update main.cpp

New questions :) Small bug fixes