Added controls and death/restart screens