Add win/lose screen -> test, screen call back, control