Update to work with Knockout 3.1 (but still backwards compatible)