Removed some extra code
Removed some extra code and added comments