Updated to use HEAD