fourth commit
Just added the student count
Adding comments to the file
refactoring the code