Add team enum and apply it to square