modified init.rb (not tested)