Trying to add the success message model