update code - remove debug code