Updated Main to match lesson