vignesh_p's submissions for second problem