Small changes after test game