Fix draw_a_square_custom