Add test specs to problem 1