Added test for switch_layer
Fix Lint