Adding assignment 4