Finish Part three : Assignment 6