Colocar uma esperar no Java usando Selenium

Olá pessoal,

Sou nova por aqui e na programação e necessito da ajuda de vocês.
Estou estudando automação e usando o site da apple para realizar meus testes.
Criei um código que deve realizar a compra do iphone 8 plus, mas ao chegar na página para seleção do modelo através da automação, a automação simplesmente pára, acredito que o comando click está tentando clicar antes da página ser totalmente carregada e por isso dá o erro.
Como inserir um pause, break, wait, algum comando que faça o código aguardar o carregamento para só depois realizar o clique?
Para tal ação vou precisar fazer algum tipo de import?

Thread.sleep(milessegundos) faz a Thread principal aguardar um certo tempo, apesar de não ser o mais correto (definir um tempo de espera), funciona…

O certo seria esperar até o carregamento terminar, mas não me recordo como.

Pesquise por: selenium wait page load, que você deverá achar como faz