Added primary keys to the test function in main.cpp