Augmenting Knockout JS