final comments

A few finishing touches
added error check

added a quick while loop check to look for valid input