Working on model controller