Updated main.cpp to v0.97

From 0.96 to 0.97