add option for custom class