Added problem 01 tests