Change model to handle story sort