Add unit tests for initial state