fix q04 method to pass unit test
done with q07