Add the virtual_attributes through including the module VirtualAttributes
Replace the method setup_virtual_attributes by a module VirtualAttributes