Fixed 2 small errors