Python Array Set Size Of Screen