updates to reflect changes in code