resolved small recommendation issue
Added simple test