Adding custom class