Assignment 1 Problem A Solution