Update gen_square01.rb