Add input validation