handle incorrect input better

Note: this is just prototype grade...we should handle things better.