Adding assignment 6