Переписал под свой лад. Я так понимаю вы этого эффекта хотели?
<button id="mainbutton">Создать фигуру</button>
<script>
let colors = ['red', 'green', 'blue'];
$(document).on('click', '#mainbutton', function(){
let div = document.createElement('div')
let size = Math.floor((Math.random()*100)+5);
let randomColor = colors[Math.floor(Math.random()*colors.length)]
$(div).css("width", size+"px")
$(div).css("height", size+"px")
$(div).css('backgroundColor', randomColor)
document.body.append(div)
});
</script>
