Updated to reflect new way to use generator