Add code from class example