Driver states
Adding capabilities and test_capabilities to unit tests