Added custom class stuff.