Schools focus on mental health as students return to class