Tweaks to Student and Teacher models