Missing states issue fixed

Resolves #3.