Modified - main.cpp
Fix #4