Please help me to correct it with python

$[88]$: import random

def displaymatrix$($n$)$:

matrix $=[[$random$.$randint$(0,1)$ for $\_$ in range$($n$)]$ for $\_$ in range$($n$)]$

return matrix

def printmatrix$($matrix$)$:

for i in range$($matrix$)$:

for j in range$($matrix$)$:

print$($random$.$randint$(0,1),$ end$=\text{'}\text{'})$

print$()$

def main$()$:

size $=$ int$($input$("$Enter the size of the matrix : $"))$

print$($matrix$)$

if $\_\_$name$\_\_=="\_\_$main$\_\_"$:

main$()$

Enter the size of the matrix : $4$

$[]$

$```$