This was my solution of my friend's problem