python stop and wait program