Update entity.rb