Added code from week2 to Week5