Added needed modules
Mark checkIfCanRegistration and unknownState as private