Added wait function (untested), screenshot