Re-introduce the waiting loop for input grabbing