Adding support for ActiveRecord