Re-use getBlockAt and rename squareBlockAt to flatBlockAt