Adding some unit tests for  the third homework.