Add input functions to defscreen