Update data_collection.rb