Add solution for stage-2