CS2211 Code Assignment 4