variable class rather than hardcoded entity