End game now works
More work towards flag elements