Added to main.cpp to test