Updated extension