Add initial state layer documentation