added edit fn