Refactored to put students in array