Trying adding callback