Finished problem set #2