Add array implementation
Write test