Adding code for recitation 2