Wrote additional test for end game