Adding solution for 3-6