Update code for drought layers