Implement indexed take as well