Update TODO, UI changes for victory screen