Adding code w/ Problem1 Solution