Unit test updated and typo corrected