Add code done in recitation 6