adding assignment 2