add a class to handle data mapping