Update main.rb

Delete p function