End game half working