Implement functionality for jagged_array