-added take-home problem code