Minor update because of removed prints