Updated tests/src/main.cpp