Add test classes for problem 1, comments to problem 1 and 2 files