small updates to be consistent w/ code