Add a method to update the screen size