Add args to load data method