Added model for demo