2d array ds hackerrank python