Updates to week2 problems