Add shebang and a simpler wrapper function

This is just the result of me making your .py script usable by myself when playing with it ;-).