Refactor of the test stuff to be a bit more flexible, still janky, but better. Renamed MainLayer to ExampleLayer, more fitting.