Week 3 Problem A & B