Update documentation, improve using array