Add code to return class name