Fix tests, remove debug puts and update example