Add update function for working layer