Update main.cpp
Merge pull request #2 from paulhewitt/cole

Update main.cpp