Returns the provided message in square code

Square Code

const squareCode = function(message) {
  let codedMessage = '';
  let unspacedMessage = message.replaceAll(' ', '');
  let messageLength = unspacedMessage.length;
  let cols = Math.ceil(Math.sqrt(messageLength));
  let rows = Math.ceil(messageLength / cols);
  for (let i = 0; i < cols; i++){
    for (let j = 0; j < rows; j++){
      if ((i+cols*j) < messageLength){
        codedMessage += unspacedMessage[i+cols*j];
      }
    }
    codedMessage += ' ';
  }
  return codedMessage;
}