Finish Part one : Assignment 6