Updated the readme and corrected a comment