Add input functionality