Initial add for solution for problem two