control screen and end screen added