updated code
Adding comments