Add wait method for Docker builds