Handling head updation part better