tiny comment change
tiny comment change
small bug fix and new comments added
better code structure