Updating readme and adding comments