Updated test_screen to avoid use of flat_button