Made IB use custom class