Add codes for assignment 1 and a.