Completed the assignment01 with the requested modifications