Added buttons to end screen state