Adding assignment 2