West Lothian baby graves to be protected from flooding after