Replaced try/catch with checking for count
Merge pull request #3 from citizenmatt/prompt_for_restart

Replaced try/catch with checking for count