Update main.c
Merge pull request #2 from bjohanson/TOM

Update main.c