Update to reflect changes