Updating the model, adding initializer and sample data