Fixed a typo, small improvements