Fixed logic of add layer / remove layer