Menu Display Function Updates

class_name is now working and you can specify a wrapper element.