Python code for new screen