Updated repository per TA's comments