My solutions for Karan Goel's python project