Wrote on the problem statement