Add test for homework 01