Update to code -- new class created