Update Screen Function Code In Python