Updated wait to wait_for_element_displayed