Inserted missing return.
Added tests.
Added missing word to description.