Adding additional functions
improving data loader