Add comment to main.cpp
Refactor code
Continue refactoring