Solved in python, answer the questions in comments must | Chegg.com