Added code from previous homework