Refactor model to be LXModel subclasses