[ADD] Unit. testing