Modified the programming assignment