Refactor code to add three players