Tidied up directory.rb with student_count variable