Added method to initialize qubit states