I have a few todo improvements, but if you'd like to see this one tested/implemented, add as an issue on the github!