It looks like xpyb.wrap works great but it doesn't seem to update the pref_screen properly. I assume that it's because during connection is when you actually get the screen information. Is there any other way to get the default screen?