update for improved engine code