Fixed typo
Added tests