Refactor script into a class & enable positional argument passing

Closes #1 (closed). cc @weimeng if you're interested. This resolves the 2 suggestions you had in !1 (merged)