func display() {
    for i in 0..<rows {
        for j in 0..<cols {
            print(group[i*cols+j], separator: " ", terminator: " ")
        }
        print()
    }
}