Small update to the  method