Validate input and add setDigits function