add return check generator