Minor design update