Adding different models for customBlock