Redone first two problems w/ Ruby