added the unittest class i've made check the solution