Refactored student and teacher models into one CodeFellow model