Added tests to first homework