Edited the model section