add accidentally removed lines back