Problem 2 simple solution added, golden solution will be added later