Unit tests added,
task solved