Small Python piece of code from college - View it on GitHub