Updated src/main.cpp