added custom class option