_rishi_o1's submissions for second problem