add assignment 5 and 6