Add input handling