Minor update to previous script