update solution

updating code combat solution