Updated with class example