Use traits class to communicate return types