more minor changes to refactor

Signed-off-by: Vince Montalbano <vince.montalbano@gmail.com>