Add class parameter to make custom repo optional