Minor change to print code