Problem 004 solved and commented