adding assignment 6 for EC201