Updated Classes in Custom Classes Example