Add area and radius stage sampling function