Doesn't Python already do this? https://www.youtube.com/watch?v=shQtrn1v7sQ