Add input checking