Assignment 01 Problem A Solution