Do TDD better

Instead of testing just the command line, test the methods ("units") of the Student class.