Add @coffeetocode's suggested patch to address #7.