Sample Test A: problem 01