Added code from week 4