That sounds great! I'm currently working on adding support for .py files as well – open to PRs :)