Added test capacity to each screen. Just run python name_of_the_screen.py