Add documentation to main.cpp