Help with importing id names

[Motorhomer14]

I am following a course where the instructor has import the GUI component id's without having to declare them. The original long version of declaring the id into variables is shown below.

val btn_click_me = findViewById(R.id.button) as Button 
 
btn_click_me.setOnClickListener {
     Toast.makeText(this@MainActivity, "You clicked me.", Toast.LENGTH_SHORT).show()
}

The short version replaces Val with button id declared within the bain listener line and then imported. When the instructor clicks on the button he gets the option to import this id into the code but I only get the option to declare it as a variable.

button.setOnClickListener {
    Toast.makeText(this@MainActivity, "You clicked me.", Toast.LENGTH_SHORT).show()
}

Within my activity_main.xml file I have the button id as button
android:id="@+id/button"

wonder if this is something I am doing wrong or a feature I have not turned on