Updated code in testing block.