Clean up code
Implement option of deleting student