add assignment 3 & 4