adding assignment 5.1