Do TODO. Add type of model on creation.