Adding some logic for wait for elements