Get display info from pygame instead of hard-coding it.